<?php
/**
 * @title 面试题 17.14. 最小K个数 - 堆排序
 * @author start2004
 */

// ini_set("memory_limit", "128M");

// class Solution {
//
//     function find() {
//
//     }
// }

class Solution {

    /**
     * @param Integer[] $arr
     * @param Integer $k
     * @return Integer[]
     */
    function smallestK($arr, $k) {
        /**
         * @since 2021-09-03 长度
         */
        $len = count($arr);
        $k2 = $len-$k;

        /**
         * @since 2021-09-03 0个元素
         */
        if($k == 0){
            return [];
        } else {}

        /**
         * @since 2021-09-03 所有元素
         */
        if($k == $len){
            return $arr;
        } else {}

        /**
         * @since 2021-09-03 k个元素集合
         */
        $result = [];

        /**
         * @since 2021-09-03 决策方案
         */
        if($k <= $k2){
            /**
             * @since 2021-09-03 最大堆
             */
            $maxHeap = new SplMaxHeap();

            /**
             * @since 2021-09-03 k个元素进最大堆
             */
            for ($i=0; $i<$k; $i++){
                $maxHeap->insert($arr[$i]);
            }

            /**
             * @since 2021-09-03 其他元素依次和最大堆top元素比较
             */
            for( ; $i<$len; $i++){
                /**
                 * @since 2021-09-03 比top小的元素进堆, 移出top
                 */
                if($arr[$i] < $maxHeap->top()){
                    $maxHeap->extract();
                    $maxHeap->insert($arr[$i]);
                } else {}
            }

            /**
             * @since 2021-09-03 返回最大堆的元素
             */
            for($i=0; $i<$k; $i++){
                $result[] = $maxHeap->current();
                $maxHeap->next();
            }
        } else {
            /**
             * @since 2021-09-03 反向查找, 最小堆
             */
            $minHeap = new SplMinHeap();

            /**
             * @since 2021-09-03 k2个元素进最小堆
             */
            for($i=0; $i<$k2; $i++){
                $minHeap->insert($arr[$i]);
            }

            /**
             * @since 2021-09-03 其他元素依次和最小堆top元素比较
             */
            for( ; $i<$len; $i++){
                /**
                 * @since 2021-09-03 比top大的元素进堆, 移出top
                 */
                if($arr[$i] > $minHeap->top()){
                    $result[] = $minHeap->extract();
                    $minHeap->insert($arr[$i]);
                } else {
                    $result[] = $arr[$i];
                }
            }

            /**
             * @since 2021-09-03 result里元素就是小于k2个最大元素的k个最小元素
             */
        }

        /**
         * @return
         */
        return $result;
    }
}

/**
 * @url http://127.0.0.1/leetcode/202109/2021.09.03.php
 */
$datas = [
    [[1,3,5,7,2,4,6,8], 4], // 1,2,3,4
    [[1,3,5,7,2,4,6,8], 6], // 1,2,3,4,5,6
];

include_once dirname(__DIR__) . DIRECTORY_SEPARATOR ."xhprof.php";
$xhprof = new Xhprof();
foreach ($datas as $data){
    if(count($data) <= 20){
        var_dump($data);
    } else {
        var_dump(count($data));
    }

    $obj = new Solution();
    $result = $obj->smallestK(...$data);
    // $result = $obj->($xhprof->tree($data));
    // $result = $obj->($xhprof->listNode($data));
    // $result = $obj->find(...$data);
    var_dump($result);
    // if(count($result)<=20){
    //     var_dump($result);
    // } else {
    //     var_dump(count($result));
    // }
    echo str_repeat(PHP_EOL, 3);
    echo str_repeat("<br>", 3);
}

// foreach ($datas as $data){
//     $obj = new $data[0][0](...$data[1][0]);
//
//     for ($i=1; $i<count($data[0]); $i++){
//         $func = $data[0][$i];
//         $param = $data[1][$i];
//         $result = $obj->$func(...$param);
//
//         echo $data[0][$i] ."(\"". implode(",", $data[1][$i]) ."\") ";
//             if($result === true){
//                 echo "True";
//             } elseif($result === false) {
//                 echo "False";
//             } elseif($result === null) {
//                 echo "Null";
//             } elseif(is_array($result)) {
//                 var_dump($result);
//             } else {
//                 echo $result;
//             }
//         echo PHP_EOL;
//     }
//
//     echo str_repeat(PHP_EOL, 3);
// }
$xhprof->end();
